class DataFrame_Categorical_Imputer():
def __init__(self):
print("Imputation object created")
def fit(self, data):
"""
This method will fit
impute mode value for
all missing categoriical
variables
"""
self.fill = pd.Series([data[column].\
value_counts().index[0]
if data[column].dtype == np.dtype('O') else \
data[column].mode() for column in data],
index=data.columns)
return self